package com.gy.ai.agent.platform.ollama.entity;

import com.gy.ai.agent.core.ReqOptions;
import lombok.Data;
import lombok.experimental.Accessors;

import java.io.Serializable;
import java.math.BigDecimal;

/**
 * @author LGY
 * Create by 2025/6/18 14:43
 */
@Data
@Accessors(chain = true)
public class OllamaEmbeddingReq extends ReqOptions implements Serializable {

    private String model;

    private String input;

    private Boolean truncate;

    private Modelfile options;

    private String keep_alive;

    @Data
    @Accessors(chain = true)
    public static class Modelfile implements Serializable {

        private Integer mirostat;

        private BigDecimal mirostat_eta;

        private BigDecimal mirostat_tau;

        private Integer num_ctx;

        private Integer repeat_last_n;

        private BigDecimal repeat_penalty;

        private BigDecimal temperature;

        private Integer seed;

        private String stop;

        private Integer num_predict;

        private Integer top_k;

        private BigDecimal top_p;

        private BigDecimal min_p;
    }

}
